Python: How to pass output of the map function as arguments